In [6]:
GCS_PATH = '/kaggle/input/gan-getting-started'
jpeg_dir = "/kaggle/input/more-monet"
MONET_FILENAMES = tf.io.gfile.glob(str(GCS_PATH + '/monet_tfrec/monet*.tfrec'))
PHOTO_FILENAMES = tf.io.gfile.glob(str(GCS_PATH + '/photo_tfrec/photo*.tfrec'))
JPEG_FILENAMES = "/kaggle/input/more-monet"
extra_files = tf.io.gfile.glob(str("/kaggle/input/more-monet/*.jpg"))
monet_files = tf.io.gfile.glob(str("/kaggle/input/gan-getting-started/monet_jpg/*.jpg"))
photo_files = tf.io.gfile.glob(str('/kaggle/input/gan-getting-started/photo_jpg/*.jpg'))
folders = ['Monet', 'Photo', 'Extra Monet Images']
num_images = [len(monet_files), len(photo_files), len(extra_files)]
# Set plot style
plt.style.use('seaborn')
# Create horizontal bar plot
fig, ax = plt.subplots(figsize=(16,12))
bar_colors = ['#F9C74F', '#90BE6D', '#43AA8B'] # set custom colors for each bar
ax.barh(folders, num_images, color=bar_colors)
# Customize plot
ax.grid(axis='x', alpha=0.5) # add gridlines
ax.set_xlabel('Number of Images', fontsize=12) # set x-axis label
ax.set_ylabel('Folder', fontsize=12) # set y-axis label
ax.set_title('Quantity of Images in Each Folder', fontsize=14, fontweight='bold') # set title
ax.set_xlim([0, max(num_images)*1.1]) # set x-axis limits to show all bars
ax.set_xticks(np.arange(0, max(num_images)*1.1, step=5)) # set x-axis tick marks
for i, v in enumerate(num_images):
ax.text(v+1, i, str(v), color='black', fontweight='bold') # show value labels next to bars
# Remove spines
for spine in ax.spines:
ax.spines[spine].set_visible(False)
# Remove y-tick labels
ax.tick_params(axis='y', which='both', length=0)
# Adjust layout
plt.tight_layout()
# Display plot
plt.show()
/tmp/ipykernel_13/307993937.py:17: MatplotlibDeprecationWarning: The seaborn styles shipped by Matplotlib are deprecated since 3.6, as they no longer correspond to the styles shipped by seaborn. However, they will remain available as 'seaborn-v0_8-<style>'. Alternatively, directly use the seaborn API instead.
plt.style.use('seaborn')
